import SnakeENV
import time

# 创建贪吃蛇游戏环境
env = SnakeENV.SnakeEnv()

# 游戏主循环
while True:
    # 获取当前游戏状态
    observation = env.reset()

    # 渲染游戏界面
    env.render()

    # 游戏逻辑
    done = False
    while not done:
        # 获取用户输入
        action = input("请输入动作（0: 上, 1: 下, 2: 左, 3: 右）: ")
        action = int(action)

        # 执行动作并获取下一个状态和奖励
        observation, reward, done, info = env.step(action)

        # 渲染游戏界面
        env.render()

        # 打印奖励和游戏信息
        print("奖励:", reward)
        print("游戏信息:", info)

        # 等待一段时间，控制游戏速度
        time.sleep(0.1)

    # 游戏结束，询问是否重新开始
    restart = input("游戏结束，是否重新开始？(y/n): ")
    if restart.lower() != "y":
        break

# 关闭游戏环境
env.close()
